Why am I getting TypeError: unhashable type: 'numpy.ndarray' in my Python code